import ollama


# 定义一个去ollama服务器获取响应信息的聊天功能.
def get_response(content):
    # 可以不写client对象.默认就是连接本地ollama
    return ollama.chat(model='qwen2:0.5b', messages=[{'role': 'user', 'content': content}])


# 写个主函数测试一下模块的功能
if __name__ == '__main__':
    while True:
        prompt = input("请录入你的提示词:")
        res = get_response(prompt)
        print("助手回答:", res.message.content)
